package com.wnzt.baszh.util.common;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.security.Key;
import java.security.SecureRandom;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import javax.crypto.Cipher;
import javax.crypto.CipherInputStream;
import javax.crypto.CipherOutputStream;
import javax.crypto.KeyGenerator;

/**
 * 文件加密和解密工具类
 * 
 * @author dqw
 * @date 2017-3-20
 */

public class FileEncodeAndDecode {
    Key key;

    public FileEncodeAndDecode(String str) {
        getKey(str);// 生成密匙
    }

    /**
     * 文件file进行加密并保存目标文件destFile中
     * 
     * @param file
     *            要加密的文件 如c:/test/srcFile.txt
     * @param destFile
     *            加密后存放的文件名 如c:/加密后文件.txt
     */
    public void encrypt(String file, String destFile) throws Exception {
        File orfile = new File(file);
        Cipher cipher = Cipher.getInstance("DES");
        // cipher.init(Cipher.ENCRYPT_MODE, getKey());
        cipher.init(Cipher.ENCRYPT_MODE, this.key);
        InputStream is = new FileInputStream(file);
        OutputStream out = new FileOutputStream(destFile);
        CipherInputStream cis = new CipherInputStream(is, cipher);
        byte[] buffer = new byte[1024];
        int r;
        while ((r = cis.read(buffer)) > 0) {
            out.write(buffer, 0, r);
        }
        cis.close();
        is.close();
        out.close();
    }

    /**
     * 对文件进行加密，加密后文件仍保持原文件名
     * 
     * @param file
     *            要加密的文件路径 如c:/test/srcFile.txt
     * @throws Exception
     */
    public void encrypt(String file) throws Exception {
        // 加密后文件的临时保存文件 名
        String destFile = file.substring(0, file.lastIndexOf(".")) + "_copy" + file.substring(file.lastIndexOf("."));
        encrypt(file, destFile); // 对文件加密

        // 加密完成后， 将文件修改成原文件名
        File enFile = new File(destFile);
        File afterFile = new File(file);
        enFile.renameTo(afterFile);
    }

    /**
     * 文件采用DES算法解密文件
     * 
     * @param file
     *            已加密的文件 如c:/加密后文件.txt * @param destFile 解密后存放的文件名 如c:/
     *            test/解密后文件.txt
     */
    public void decrypt(String file, String dest) throws Exception {
        File enfile = new File(file);
        String path = dest.substring(0, dest.lastIndexOf("\\") + 1);
        File copyPath = new File(path);
        if (!copyPath.exists()) {
            copyPath.mkdirs();
        }
        Cipher cipher = Cipher.getInstance("DES");
        cipher.init(Cipher.DECRYPT_MODE, this.key);
        InputStream is = new FileInputStream(file);
        OutputStream out = new FileOutputStream(dest);
        CipherOutputStream cos = new CipherOutputStream(out, cipher);
        byte[] buffer = new byte[1024];
        int r;
        while ((r = is.read(buffer)) >= 0) {
            cos.write(buffer, 0, r);
        }
        cos.close();
        out.close();
        is.close();
        /*
         * enfile.delete(); //解密成功后删掉加密的文件，只保存解密后的文件 File enFile = new
         * File(dest); File afterFile = new File(file);
         * enFile.renameTo(afterFile);
         */
    }

    /**
     * 根据参数生成KEY
     */
    public void getKey(String strKey) {
        try {
            KeyGenerator _generator = KeyGenerator.getInstance("DES");
            _generator.init(new SecureRandom(strKey.getBytes()));
            this.key = _generator.generateKey();
            _generator = null;
        } catch (Exception e) {
            throw new RuntimeException("Error initializing SqlMap class. Cause: " + e);
        }
    }
    public static void main(String[] args) {
		try {
			List<File> filelist=new ArrayList<File>();
			filelist=PrintUtils.getFileList(filelist,"E:\\pdf2");
			for (File file : filelist) {
				new FileEncodeAndDecode("wnzt.baszh.archive.key").encrypt(file.getPath(), "E:/pdf/"+file.getName());
			}
			
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}
